#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @author: 123
# @time: 2023/04/20 14:22:47

import os
import sys

def check_cpu_usage():
    '''
    检查 CPU 使用情况
    '''
    cmd = "top -d 1 -b -n 2 | grep 'Cpu(s)' | tail -n 1 | awk '{print $2+$4}'"
    res = os.popen(cmd).readlines()
    cpu_usage = float(res[0].strip())
    print("CPU 使用率: %s%%" % cpu_usage)
    if cpu_usage > 90:
        print("警告: CPU 使用率过高，请及时处理")

def check_memory_usage():
    '''
    检查内存使用情况
    '''
    cmd = "free -m | grep 'Mem' | awk '{print $3/$2 * 100.0}'"
    res = os.popen(cmd).readlines()
    memory_usage = float(res[0].strip())
    print("内存使用率: %s%%" % memory_usage)
    if memory_usage > 90:
        print("警告: 内存使用率过高，请及时处理")

if __name__ == '__main__':
    check_cpu_usage()
    check_memory_usage()